# 初始化
factory = AIFactory()

# 创建基础数字人视频
result = factory.create_digital_human(
    text="大家好，欢迎观看这个数字人演示视频",
    provider="heygen",
    avatar_id="male1"
)

if result["success"]:
    print(f"任务创建成功，任务ID: {result['task_id']}")

    # 检查状态
    status = factory.check_dh_status("heygen", result["task_id"])
    if status["status"] == "completed":
        print(f"视频生成完成: {status['video_url']}")

# 创建场景数字人视频
scene_result = factory.create_scene_video(
    scene_type="education",
    script="机器学习是人工智能的重要分支，它让计算机能够从数据中学习规律。"
)

# 批量生成不同场景
scripts = {
    "presentation": "今天我将介绍我们的新产品",
    "education": "这个数学公式的推导过程如下",
    "customer_service": "请问有什么可以帮您的吗？"
}

batch_results = factory.dh_scene.batch_create_scenes(scripts)